package com.c2b.algorithm.leetcode.base;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * <a href="https://leetcode.cn/problems/find-all-anagrams-in-a-string/">找到字符串中所有字母异位词(Find All Anagrams in a String)</a>
 * <p>给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。</p>
 * <p>异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。</p>
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1:
 *      输入: s = "cbaebabacd", p = "abc"
 *      输出: [0,6]
 *      解释:
 *          起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
 *          起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。
 *
 *  示例 2:
 *      输入: s = "abab", p = "ab"
 *      输出: [0,1,2]
 *      解释:
 *          起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。
 *          起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。
 *          起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。
 * </pre>
 * </p>
 * <p>
 * <b>提示：</b>
 * <ul>
 *     <li>1 <= s.length, p.length <= 3 * 10^4</li>
 *     <li>s 和 p 仅包含小写字母</li>
 * </ul>
 * </p>
 *
 * @author c2b
 * @since 2024/8/19 17:28
 */
public class LC0438FindAllAnagramsInString_M {
    static class Solution {
        public List<Integer> findAnagrams(String s, String p) {
            List<Integer> resultList = new ArrayList<>();
            int sLen = s.length();
            int pLen = p.length();
            if (sLen < pLen) {
                return resultList;
            }
            // 初始化一个数组来统计字符串 p 中每个字符的出现次数
            int[] sCnt = new int[26];
            int[] pCnt = new int[26];
            for (int i = 0; i < pLen; i++) {
                sCnt[s.charAt(i) - 'a']++;
                pCnt[p.charAt(i) - 'a']++;
            }
            if (Arrays.equals(sCnt, pCnt)) {
                resultList.add(0);
            }
            for (int i = pLen; i < sLen; ++i) {
                sCnt[s.charAt(i - pLen) - 'a']--;
                sCnt[s.charAt(i) - 'a']++;
                if (Arrays.equals(sCnt, pCnt)) {
                    resultList.add(i - pLen + 1);
                }
            }
            return resultList;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        Printer.printListInteger(solution.findAnagrams("cbaebabacd", "abc"));
        Printer.printListInteger(solution.findAnagrams("abab", "ab"));
    }
}
